Linux下,Nvidia驱动以及Cuda的安装 |
您所在的位置:网站首页 › Linux 驱动安装 很麻烦 › Linux下,Nvidia驱动以及Cuda的安装 |
最近刚好在看动手深度学习的书,需要在GPU环境下运行,故此记录一下。 一、确认安装环境 1.禁用nouveau(否则与NVIDIA驱动安装会有冲突,nouveau是linux自带系统驱动) 右键打开终端 sudo vi /etc/modprobe.d/blacklist.conf(没有安装vim编辑器需要先执行 sudo apt-get install vim) 在打开的文件末尾添加以下两行代码 blacklist nouveau options nouveau modeset=0 保存退出 update-initramfs –u (重新生成 kernel initramfs) 重启后(如果电脑有两个显卡驱动,图标不会变化;否则,图标会变大)打开终端输入一下命令: lsmod | grep nouveau (没有输出表示已经禁用) 2.Nvidia驱动安装A.执行如下命令将Ubuntu官方维护的NVIDIA显卡驱动软件源添加到系统中: sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get updateB.执行如下命令,查看适合自己电脑的显卡驱动,找到recommended对应项 sudo ubuntu-drivers devicesC找到电脑的设置>>软件和更新>>附加驱动>>选择对应的Nvidia驱动>>点击右下方按钮更改,完成后重启执行如下代码 nvidia-smi 测试驱动是否安装成功,成功则显示如下图:
A 安装过程选项选择 由于自己在台式机和笔记本上分别安装cuda8.0和cuda10.1。 (1)对于cuda8.0会有一个很长的阅读声明,一直按住enter键直100%;而对于cuda10.1版本直接就是个图形化界面。阅读完成后,输入accept即可。 (2)cuda8.0中,选项install nvidia accelerated Graphics Driver,输入n,因为我们已经安装了nvidia的驱动;后面的选型均输入y,等待安装即可; 对于cuda10.1在它对应图形化界面取消该项,默认是【x】,ENTER键取消,选择install安装即可(为图形界面第一个选型) 说明:无论安装什么版本,我们都不安装Graphics Driver这一项即可(填n),其余选项均为 y (3)如何最后看到有三行中,最后两行有两个installed,表示安装成功 B 输入reboot,重启电脑 C 相关配置 sudo vim ~/.bashrc 打开改文件 在文件最后添加以下两行: export PATH=/usr/local/cuda-(对应版本号,如:10.1)/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-(同上)/lib64:$LD_LIBRARY_PATH12 保存后退到终端,执行如下命令: source .bashrc nvcc -V 检查是否安装成功 3测试 执行如下代码: cd /usr/local/cuda/samples cd 1_Utilities/deviceQuery sudo make ./deviceQuery 当看到Result=pass,则表示表示测试成功,如下图所示
|
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |